package com.orange.cloud.common.utils.image;

import javax.imageio.ImageIO;
import javax.swing.*;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;

public class ImgTools {
    /**
     * 将图片压缩到指定大小以内
     *
     * @param srcImgData 源图片数据
     * @param maxSize 目的图片大小
     * @return 压缩后的图片数据
     */
    public static byte[] compressUnderSize(byte[] srcImgData, long maxSize) {
        double scale = 0.7;
        byte[] imgData = Arrays.copyOf(srcImgData, srcImgData.length);

        if (imgData.length > maxSize) {
            do {
                try {
                    imgData = compress(imgData, scale);

                } catch (IOException e) {
                    throw new IllegalStateException("压缩图片过程中出错，请及时联系管理员！", e);
                }

            } while (imgData.length > maxSize);
        }

        return imgData;
    }

    /**
     * 按照 宽高 比例压缩
     * @param scale 压缩刻度
     * @return 压缩后图片数据
     * @throws IOException 压缩图片过程中出错
     */
    public static byte[] compress(byte[] srcImgData, double scale) throws IOException {
        BufferedImage bi = ImageIO.read(new ByteArrayInputStream(srcImgData));
        int width = (int) (bi.getWidth() * scale); // 源图宽度
        int height = (int) (bi.getHeight() * scale); // 源图高度

        Image image = bi.getScaledInstance(width, height, Image.SCALE_SMOOTH);
        BufferedImage tag = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);

        Graphics g = tag.getGraphics();
        g.setColor(Color.RED);
        g.drawImage(image, 0, 0, null); // 绘制处理后的图
        g.dispose();

        ByteArrayOutputStream bOut = new ByteArrayOutputStream();
        ImageIO.write(tag, "JPEG", bOut);

        return bOut.toByteArray();
    }
    /**
     * @description
     * @param srcImgFile 源图片
     * @param waterMarkContent 水印内容
     * @param fileExt 图片格式
     * @return void
     */
    public static byte[] addWatermark(InputStream srcImgFile, int srcImgWidth, int srcImgHeight, String waterMarkContent, String fileExt){
        Font font = new Font("宋体", Font.BOLD, 100);//水印字体，大小
        Color markContentColor = new Color(255,255,255);//水印颜色
        Integer degree = 45;//设置水印文字的旋转角度
        float alpha = 0.5f;//设置水印透明度
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        try {
            Image srcImg = ImageIO.read(srcImgFile);//文件转化为图片
            // 加水印
            BufferedImage bufImg = new BufferedImage(srcImgWidth, srcImgHeight, BufferedImage.TYPE_INT_RGB);
            Graphics2D g = bufImg.createGraphics();//得到画笔
            g.drawImage(srcImg, 0, 0, srcImgWidth, srcImgHeight, null);
            g.setColor(markContentColor); //设置水印颜色
            g.setFont(font);              //设置字体
            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));//设置水印文字透明度
            if (null != degree) {
                g.rotate(Math.toRadians(degree));//设置水印旋转
            }
            JLabel label = new JLabel(waterMarkContent);
            FontMetrics metrics = label.getFontMetrics(font);
            int width = metrics.stringWidth(label.getText());//文字水印的宽
            int rowsNumber = srcImgHeight/width;// 图片的高  除以  文字水印的宽    ——> 打印的行数(以文字水印的宽为间隔)
            int columnsNumber = srcImgWidth/width;//图片的宽 除以 文字水印的宽   ——> 每行打印的列数(以文字水印的宽为间隔)
            //防止图片太小而文字水印太长，所以至少打印一次
            if(rowsNumber < 1){
                rowsNumber = 1;
            }
            if(columnsNumber < 1){
                columnsNumber = 1;
            }
            rowsNumber=rowsNumber*2;
            for(int j=0;j<rowsNumber;j++){
                for(int i=0;i<columnsNumber;i++){
                    g.drawString(waterMarkContent, i*width + j*width, -i*(width/2) + j*(width/2));//画出水印,并设置水印位置
                }
            }
            g.dispose();// 释放资源
            // 输出图片
            ImageIO.write(bufImg, fileExt, out);
            return out.toByteArray();
        } catch (Exception e) {
            e.printStackTrace();
        } finally{
            try {
                if(out != null){
                    out.flush();
                    out.close();
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }

    /**
     * @desc 为图片设置水印（标题、时间）
     * @date 2019/08/08
     * @author xukang
     */
    public static byte[] addWatermarkNew(InputStream srcImgFile, int srcImgWidth, int srcImgHeight, String waterMarkContent, String fileExt){
        Image srcImg;
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        try {
            srcImg = ImageIO.read(srcImgFile);
            BufferedImage image = new BufferedImage(srcImgWidth, srcImgHeight, BufferedImage.TYPE_INT_RGB);
            // 获取Graphics2D
            Graphics2D g2d = image.createGraphics();
            g2d.drawImage(srcImg, 0, 0, srcImgWidth, srcImgHeight, null);
            // 画图
            g2d.setBackground(new Color(255, 255, 255));
            // 抗锯齿
            g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
            //设置水印文字透明度
            g2d.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, 0.8f));
            g2d.setPaint(new Color(0, 0, 0));
            g2d.setColor(Color.red);
            computeFontSize(g2d, srcImgWidth, srcImgHeight, 150, waterMarkContent);

            FontMetrics fontMetrics = g2d.getFontMetrics();
            int textWidth = fontMetrics.stringWidth(waterMarkContent);
            int widthX = (srcImgWidth - textWidth) / 2;
            int widthY = srcImgHeight / 2;
            g2d.rotate(Math.toRadians(45), srcImgWidth / 2, srcImgHeight / 2);
            // 表示这段文字在图片上的位置(x,y) .第一个是你设置的内容。
            g2d.drawString(waterMarkContent, widthX, widthY);
            String nowStr = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss").format(new Date());
            //时间文字长度
            int timeStrWidth = fontMetrics.stringWidth(nowStr);
            g2d.drawString(nowStr, widthX + (textWidth - timeStrWidth) / 2, widthY + fontMetrics.getHeight() + 5);
            // 释放对象
            g2d.dispose();
            // 保存文件
            ImageIO.write(image, fileExt, out);
            return out.toByteArray();
        } catch (Exception e) {
            e.printStackTrace();
        } finally{
            try {
                if(out != null){
                    out.flush();
                    out.close();
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }

    /**
     * @desc 避免因图片尺寸不一致  水印文字超出边界的问题。让文字适应图片大小做调整
     * @date 2019/8/8
     * @param  * @g2d           画布对象
     * @param  * @imgWidth      图片宽
     * @param  * @imgHeight     图片高
     * @param  * @fontSize      水印文字大小
     * @param  * @textContent   水印文字
     * @author xukang
     */
    private static void computeFontSize(Graphics2D g2d, int imgWidth, int imgHeight, int fontSize, String textContent){
        Font font = new Font("宋体", Font.PLAIN, fontSize);
        g2d.setFont(font);
        // 计算文字长度，计算居中的x点坐标
        FontMetrics fm = g2d.getFontMetrics(font);
        // 水印文字45°旋转后的垂直（或横向）长度
        int textSideLength = (int)(fm.stringWidth(textContent) * Math.pow(2, 0.5) * 0.5);
        //15是最小文字大小，避免太小看不见； 50为预留文字margin； 当文字横向或者纵向超出图片边界时，递归调整文字大小
        if (fontSize > 15 && (textSideLength > imgWidth - 50 || textSideLength > imgHeight - 50)){
            computeFontSize(g2d, imgWidth, imgHeight, --fontSize, textContent);
        }
    }

}
